package exercises.leetcode;

import java.util.ArrayList;
import java.util.List;

/**
 * <a href="https://leetcode-cn.com/problems/combinations/">
 * 77. 组合</a>
 *
 * <p><span color="#87CEEB">描述:</span>
 * 给定两个整数 n 和 k，返回范围 [1, n] 中所有可能的 k 个数的组合。
 * <p>
 * 你可以按 任何顺序 返回答案。
 * <p>
 *  
 * <p>
 * 示例 1：
 * <p>
 * 输入：n = 4, k = 2
 * 输出：
 * <pre>{@code
 * [
 *   [2,4],
 *   [3,4],
 *   [2,3],
 *   [1,2],
 *   [1,3],
 *   [1,4],
 * ]
 * }</pre>
 * 示例 2：
 * <p>
 * 输入：n = 1, k = 1
 * 输出：[[1]]
 *  
 * <p>
 * 提示：
 * <li>1 <= n <= 20</li>
 * <li>1 <= k <= n</li>
 *
 * @author or2
 * @date 2021年09月29日 时间: 9:25
 */
public class No77 {

    private List<List<Integer>> res;
    private int length;

    public List<List<Integer>> combine(int n, int k) {
        res = new ArrayList<>();
        length = n;
        List<Integer> combination = new ArrayList<>();

        /* 剪枝 */
        for (int i = 0; i < n - k + 1; i++)
            combine(i, k, combination);

        return res;
    }

    private void combine(int startIndex, int restNum, List<Integer> combination) {
        restNum--;
        combination.add(startIndex + 1);

        /* 终止条件 */
        if (restNum == 0)
            res.add(new ArrayList<>(combination));
        /* 剪枝 */
        else if (length - startIndex - 1 >= restNum)
            for (int i = startIndex + 1, LEN = length; i < LEN; i++)
                combine(i, restNum, combination);

        /* 回溯 */
        combination.remove(combination.size() - 1);
    }
}
